#include "camera.hpp"
#include "opencv2/opencv.hpp"
#include "myyolodetect.hpp"

// 引入特定的OpenCV标识符
// 引入整个cv命名空间
using namespace cv;
using namespace std;

// 打开video0设备
cv::VideoCapture cap(0); // 参数0对应/dev/video0

int camera_init(void)
{
    if (!cap.isOpened())
    {
        std::cerr << "无法打开摄像头设备" << std::endl;
        return -1;
    }

    // if (!cv::imwrite("output.jpg", frame, compression_params))
    // {
    //     std::cerr << "图像保存失败" << std::endl;
    //     return -1;
    // }

    return 0;
}

int camera_close(void)
{
    // cap.release();
    return 0;
}

std::vector<unsigned char> get_image(void)
{
    std::vector<unsigned char> buffer;

    cv::Mat frame = cv::imread("test.jpg", cv::IMREAD_COLOR);

    frame = DetectTest(frame);

    cv::imwrite("image.jpg", frame);

    // if (cap.isOpened())
    // {
    //     cv::Mat frame;
    //     cap >> frame; // 捕获一帧图像

    //     buffer.clear();

    //     if (frame.empty())
    //     {
    //         std::cerr << "未捕获到图像数据" << std::endl;
    //     }
    //     else
    //     {
    //         frame = DetectTest(frame);

    //         cv::imwrite("image.jpg", frame);
    //     }
    //     // else
    //     // {
    //     //     // 保存为JPEG格式
    //     //     std::vector<int> compression_params;
    //     //     compression_params.push_back(cv::IMWRITE_JPEG_QUALITY);
    //     //     compression_params.push_back(95); // 设置JPEG质量(0-100)

    //     //     // 将图像编码为 JPG 格式的二进制数据
    //     //     cv::imencode(".jpg", frame, buffer, compression_params);
    //     // }
    // }

    return buffer;
}
